from torchvision.models import resnet18
import torch
import os
import inspect

def get_model():
    model=resnet18(pretrained=False)
    return model

def get_input_data():
    return torch.randn(1000,3,224,224)


# def get_model():
#     return mocha.get_model()

# def get_input_data():
#     return mocha.get_input_data()  


# def call_model(model, input_data):
#     args, kwargs, padder = input_data
    
#     # 调用模型
#     output = model(*args, **kwargs)
    
#     # 如果有需要，可以进行unpad
#     if isinstance(output, torch.Tensor):
#         output = padder.unpad(output)
    
#     return output


# model = get_model()
# input_data = get_input_data()
# result = call_model(model, input_data)